#!/usr/bin/env python
# -*- coding:utf-8 -*-
# @Time      :2021/11/9 22:47
# @Author    :cjw
from use_custom_web.base.base_page import BasePage
from use_custom_web.util.file_operator import YmlOperator
from use_custom_web.setting import FEISHU_BASE_URL, ELEMENT_YML


class MainPage(BasePage):
	"""飞书主页"""

	# 添加定位元素到_locators
	BasePage._locators.update(YmlOperator(ELEMENT_YML['mp']).data)
	# 会作为用例的起始页面，就添加一个url属性，这样用例起始的时候，就可以打开对应的页面
	_base_url = FEISHU_BASE_URL

	def goto_calender(self):
		"""跳转到日历页面"""
		from use_custom_web.page.calendar_page import CalendarPage

		# 调转进入日历页面
		self.function_menu.click()
		self.menu_calendar.click()

		# 获取新的页面窗口句柄，并切换
		window_handles = self.driver.window_handles
		self.driver.switch_to.window(window_handles[-1])
		return CalendarPage(self.driver)
